package cn.ike.web.controller.system;

import cn.ike.domain.system.Module;
import cn.ike.domain.system.Role;
import cn.ike.service.system.ModuleService;
import cn.ike.service.system.RoleService;
import cn.ike.web.controller.BaseController;
import com.github.pagehelper.PageInfo;
import lombok.extern.log4j.Log4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.util.StringUtils;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RequestParam;
import org.springframework.web.bind.annotation.ResponseBody;
import org.springframework.web.servlet.ModelAndView;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

@Controller
@Log4j
@RequestMapping("/system/role")
public class RoleController extends BaseController {

    // 注入RoleService
    @Autowired
    private RoleService roleService;

    /**
     * 1.角色列表分页
     */
    @RequestMapping("list")
    public ModelAndView list(
            @RequestParam(defaultValue = "1") Integer pageNum,
            @RequestParam(defaultValue = "5") Integer pageSize) {
        ModelAndView model = new ModelAndView();
        try {
            log.info("执行角色列表分页查询开始...");
            // 调用service查询角色列表
            PageInfo<Role> pageInfo = roleService.findByPage(getLoginCompanyId(), pageNum, pageSize);
            model.addObject("pageInfo", pageInfo);
            model.setViewName("system/role/role-list");
            log.info("执行角色列表分页查询结束...");
        } catch (Exception e) {
            e.printStackTrace();
            log.error("执行角色列表分页查询出现异常！", e);
        }
        return model;
    }

    /**
     * 2.进入添加角色页面
     */
    @RequestMapping("toAdd")
    public ModelAndView toAdd() {
        ModelAndView model = new ModelAndView();
        try {
            log.info("执行添加角色页面开始...");
            model.setViewName("system/role/role-add");
            log.info("执行添加角色页面结束...");
        } catch (Exception e) {
            e.printStackTrace();
            log.error("执行添加角色页面出现异常！", e);
        }
        return model;
    }

    /**
     * 3.角色的添加或修改
     */
    @RequestMapping("edit")
    public ModelAndView edit(Role role) {
        ModelAndView model = new ModelAndView();
        role.setCompanyId(getLoginCompanyId());
        role.setCompanyName(getLoginCompanyName());
        try {
            if (StringUtils.isEmpty(role.getId())) {
                log.info("执行角色的添加开始...");
                // id为空就是添加角色
                roleService.save(role);
                log.info("执行角色的添加结束...");
            } else {
                log.info("执行角色的修改开始...");
                // 否则就是更新角色
                roleService.update(role);
                log.info("执行角色的修改结束...");
            }
            model.setViewName("redirect:/system/role/list");
        } catch (Exception e) {
            e.printStackTrace();
            log.error("执行角色的添加或修改出现异常！", e);
        }
        return model;
    }

    /**
     * 4.修改角色信息数据回显
     */
    @RequestMapping("toUpdate")
    public ModelAndView toUpdate(String roleId) {
        ModelAndView model = new ModelAndView();
        try {
            log.info("执行修改角色信息开始...");
            Role role = roleService.findById(roleId);
            model.addObject("role",role);
            model.setViewName("/system/role/role-update");
            log.info("执行修改角色信息结束...");
        } catch (Exception e) {
            e.printStackTrace();
            log.error("执行修改角色信息出现异常！", e);
        }
        return model;
    }

    /**
     * 5.删除角色
     */
    @RequestMapping("delete")
    @ResponseBody
    public Map<String, String> delete(String roleId) {
        Map<String, String> map = new HashMap<>();
        try {
            log.info("执行删除角色的开始...");
            boolean flag = roleService.delete(roleId);
            if (flag){
                map.put("msg", "删除成功！");
            } else {
                map.put("msg", "删除失败：当前删除的角色被外键引用！");
            }
            log.info("执行删除角色的结束...");
        } catch (Exception e) {
            e.printStackTrace();
            log.error("执行删除角色的出现异常！", e);
        }
        return map;
    }

    /**
     * 进入role-module.jsp角色权限页面
     */
    @RequestMapping("roleModule")
    public ModelAndView roleModule(String roleId) {
        ModelAndView model = new ModelAndView();
        try {
            log.info("执行进入角色权限页面开始...");
            Role role = roleService.findById(roleId);
            model.addObject("role", role);
            model.setViewName("system/role/role-module");
            log.info("执行进入角色权限页面结束...");
        } catch (Exception e) {
            e.printStackTrace();
            log.error("执行进入角色权限页面出现异常！", e);
        }
        return model;
    }

    @Autowired
    private ModuleService moduleService;
    /**
     * 角色分配权限（2）role-module.jsp页面异步请求，返回ztree需要的json数据
     * 返回json数据 result = [
     *     { id:2, pId:0, name:"随意勾选 2", checked:true, open:true}
     * ]
     */
    @RequestMapping("getZtreeNode")
    @ResponseBody
    public List<Map<String, Object>> getZtreeNode(String roleId){
        List<Map<String, Object>> list = new ArrayList<>();
        try {
            log.info("执行初始化zTree的开始...");
            // 查询数据：查询所有的权限
            List<Module> moduleList = moduleService.findAll();
            // 查询角色所拥有的权限
            List<Module> roleModuleList = moduleService.findModulByRoleId(roleId);
            // 封装结果
            // 遍历所有权限
            for (Module module : moduleList) {
                Map<String, Object> map = new HashMap<>();
                // 注意：map的key一定要与ztree需要的json格式一致
                map.put("id", module.getId());
                map.put("pId", module.getParentId());
                map.put("name", module.getName());
                map.put("open", true);
                // 如果角色权限列表中有所有权限中对应的权限，那么就选中
                if (roleModuleList.contains(module)) {
                    map.put("checked",true);
                }
                // 添加到返回的集合
                list.add(map);
            }
            log.info("执行初始化zTree的()结束...");
        } catch (Exception e) {
            e.printStackTrace();
            log.error("执行初始化zTree的出现异常！", e);
        }
        list.forEach(System.out::println);
        return list;
    }

    /**
     * 角色分配权限（3）role-module.jsp点击保存
     * 请求参数：
     *    roleId    角色id（一个）
     *    moduleIds 权限id（多个，逗号隔开）
     */
    @RequestMapping("updateRoleModule")
    public ModelAndView updateRoleModule(String roleId, String moduleIds) {
        ModelAndView model = new ModelAndView();
        try {
            log.info("执行角色权限保存的开始...");
            moduleService.updateRoleModule(roleId, moduleIds);
            model.setViewName("redirect:/system/role/list");
            log.info("执行角色权限保存的结束...");
        } catch (Exception e) {
            e.printStackTrace();
            log.error("执行角色权限保存的出现异常！", e);
        }
        return model;
    }
}
